# -*- coding: utf-8 -*-

class config:
    manualseed = 0

    # 数据集参数
    num_workers = 2
    batch_size = 64


    # 训练参数
    num_epochs = 10
    start_epoch = 1
    lr = 0.1
    momentum = 0.9
    weight_decay = 0.0001
    checkout = "checkout"
    pretrained = ""
    log_interval = 100
    test_interval = 20
    is_best = 0.
    # train prune
    sr = False # BN
    s = 1e-4 # BN rate
    best_model = "{}/best_model.sr={}.pth".format(checkout, sr)

    # prune setting
    prune_bn_rate = 0.7
    prune_model = "{}/prune_model.pth".format(checkout)

    # finetune
    prune_finetune = "{}/prune_finetune.pth".format(checkout)


